/******************************************************************************
 * Copyright (C) 2015  671643387@qq.com
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 *****************************************************************************/

#ifndef SYSTEM_LOCKGUARD_H_
#define SYSTEM_LOCKGUARD_H_

#if defined(_MSC_VER) && (_MSC_VER >= 1200)
# pragma once
#endif

#include "node/node.h"
#include "node/assert.h"

namespace node {
    namespace system {

        template<typename T>
        class LockGuard
        {
        public:
            LockGuard(T &lock) : lock_(lock) { lock_.lock(); }
            ~LockGuard(void) { lock_.unlock(); }
            bool isLocked(void) const { return lock_.isLocked(); }
        private: 
            T &lock_;
            DISALLOW_COPY_AND_ASSIGN(LockGuard);
        };

    }
}

#define NODE_GUARD_LOCK(Var)                                            \
    ::node::system::LockGuard<::node::system::Mutex> guard##Var(Var);   \

#define NODE_GUARD_LOCK_ASSERT(Var)                                     \
    ::node::system::LockGuard<::node::system::Mutex> guard##Var(Var);   \
    if (!guard##Var.isLocked()) { NODE_ASSERT(false); }                 \

#define NODE_GUARD_LOCK_RETURN(Var)                                     \
    ::node::system::LockGuard<::node::system::Mutex> guard##Var(Var);   \
    if (!guard##Var.isLocked()) { return; }                             \

#define NODE_GUARD_LOCK_RETURN_VAR(Var, Rtn)                            \
    ::node::system::LockGuard<::node::system::Mutex> guard##Var(Var);   \
    if (!guard##Var.isLocked()) { return Rtn; }                         \

#define NODE_GUARD_LOCK_ASSERT_RETURN(Var)                              \
    ::node::system::LockGuard<::node::system::Mutex> guard##Var(Var);   \
    if (!guard##Var.isLocked()) { NODE_ASSERT(false); return; }         \

#define NODE_GUARD_LOCK_ASSERT_RETURN_VAR(Var, Rtn)                     \
    ::node::system::LockGuard<::node::system::Mutex> guard##Var(Var);   \
    if (!guard##Var.isLocked()) { NODE_ASSERT(false); return Rtn; }     \

#endif // SYSTEM_LOCKGUARD_H_
